iT邦幫忙

2023 iThome 鐵人賽

DAY 17
0
AI & Data

深度學習-TensorFlow系列 第 17

DAY17-TensorFlow常用模組(上)

  • 分享至 

  • xImage
  •  

今天我們來介紹一下TensorFlow常會使用到的幾種模組。/images/emoticon/emoticon06.gif

1. tf.train.Checkpoint : 變數的保存與還原

tf.train.Checkpoint() 能夠接受的初始化參數比較特殊,是一個** kwargs。也就是一系列的鍵值對,鍵名沒有限制,值為一個需要保護得對象。
https://ithelp.ithome.com.tw/upload/images/20231002/20162982KNLbOmeFfu.png

模型訓練完成需要保存時可以使用:
https://ithelp.ithome.com.tw/upload/images/20231002/20162982Glg9vOC946.png
(save_path_with_prefix 是儲存檔案的目錄+前綴)

還原模型變數:
https://ithelp.ithome.com.tw/upload/images/20231002/20162982bQdKJshbTV.png
(save_path_with_prefix_and_index 是之前儲存文件的目錄+前綴+編號)

2. TensorBoard : 訓練過程可視化
在代碼目錄下建立一個名為tensorboard的資料夾來存放TensorBoard的紀錄文件,並實例化一個紀錄器。
https://ithelp.ithome.com.tw/upload/images/20231002/20162982vWiggWcQn3.png

當需要紀錄訓練過程中的參數時 :
https://ithelp.ithome.com.tw/upload/images/20231002/20162982ed2NZ67H50.png
(每運行一次tf.summary.scalar(),紀錄器就會像紀錄檔案中寫入一條紀錄。)

當需要對訓練過程可視化時,在代碼目錄打開終端:
https://ithelp.ithome.com.tw/upload/images/20231002/20162982u7tVSSQwId.png
(如若需要可進入TensorFlow的conda環境)

https://ithelp.ithome.com.tw/upload/images/20231002/20162982VbkP3WVGFg.png
(圖片擷取至網路)
(可透過右上方按鈕更新資訊,預設是30秒自動更新一次資料)

剩下還有幾種,我們明天繼續來介紹~


上一篇
DAY16-LSTM實作(下)
下一篇
DAY18-TensorFlow常用模組(中)
系列文
深度學習-TensorFlow30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言